equations with python